Passing the name of a pandas dataframe column to a function